#ifndef RAPIDXML_ITERATORS_HPP_INCLUDED
#define RAPIDXML_ITERATORS_HPP_INCLUDED

// Copyright (C) 2006, 2009 Marcin Kalicinski
// Version 1.13
// Revision $DateTime: 2009/05/13 01:46:17 $
// ! \file rapidxml_iterators.hpp This file contains rapidxml iterators

#include "rapidxml.hpp"

namespace rapidxml
{
	// ! Iterator of child nodes of xml_node
	template<class Ch>
	class node_iterator
	{
public:

		typedef typename xml_node<Ch> value_type;
		typedef typename xml_node<Ch> &reference;
		typedef typename xml_node<Ch> *pointer;
		typedef std::ptrdiff_t difference_type;
		typedef std::bidirectional_iterator_tag iterator_category;

		node_iterator()
			: m_node( 0 )
		{}

		node_iterator( xml_node<Ch> *node )
			: m_node( node->first_node())
		{}

		reference operator *() const
		{
			assert( m_node );
			return *m_node;
		}

		pointer operator->() const
		{
			assert( m_node );
			return m_node;
		}

		node_iterator& operator++()
		{
			assert( m_node );
			m_node = m_node->next_sibling();
			return *this;
		}

		node_iterator operator++( int )
		{
			node_iterator tmp = *this;
			++this;
			return tmp;
		}

		node_iterator& operator--()
		{
			assert( m_node && m_node->previous_sibling());
			m_node = m_node->previous_sibling();
			return *this;
		}

		node_iterator operator--( int )
		{
			node_iterator tmp = *this;
			++this;
			return tmp;
		}

		bool operator ==( const node_iterator<Ch> &rhs )
		{
			return m_node == rhs.m_node;
		}

		bool operator !=( const node_iterator<Ch> &rhs )
		{
			return m_node != rhs.m_node;
		}

private:

		xml_node<Ch> *m_node;
	};

	// ! Iterator of child attributes of xml_node
	template<class Ch>
	class attribute_iterator
	{
public:

		typedef typename xml_attribute<Ch> value_type;
		typedef typename xml_attribute<Ch> &reference;
		typedef typename xml_attribute<Ch> *pointer;
		typedef std::ptrdiff_t difference_type;
		typedef std::bidirectional_iterator_tag iterator_category;

		attribute_iterator()
			: m_attribute( 0 )
		{}

		attribute_iterator( xml_node<Ch> *node )
			: m_attribute( node->first_attribute())
		{}

		reference operator *() const
		{
			assert( m_attribute );
			return *m_attribute;
		}

		pointer operator->() const
		{
			assert( m_attribute );
			return m_attribute;
		}

		attribute_iterator& operator++()
		{
			assert( m_attribute );
			m_attribute = m_attribute->next_attribute();
			return *this;
		}

		attribute_iterator operator++( int )
		{
			attribute_iterator tmp = *this;
			++this;
			return tmp;
		}

		attribute_iterator& operator--()
		{
			assert( m_attribute && m_attribute->previous_attribute());
			m_attribute = m_attribute->previous_attribute();
			return *this;
		}

		attribute_iterator operator--( int )
		{
			attribute_iterator tmp = *this;
			++this;
			return tmp;
		}

		bool operator ==( const attribute_iterator<Ch> &rhs )
		{
			return m_attribute == rhs.m_attribute;
		}

		bool operator !=( const attribute_iterator<Ch> &rhs )
		{
			return m_attribute != rhs.m_attribute;
		}

private:

		xml_attribute<Ch> *m_attribute;
	};
}

#endif
